package com.obggtc.picture.service.ai;

import com.obggtc.picture.enums.AiProviderEnum;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * AI服务提供商工厂
 * 负责管理和获取各种AI服务实现
 * 
 * @author system
 * @date 2025/09/29
 */
@Slf4j
@Component
public class AiProviderFactory {
    
    @Autowired
    private List<AiProvider> aiProviders;
    
    private Map<AiProviderEnum, AiProvider> providerMap = new HashMap<>();
    
    @PostConstruct
    public void initProviders() {
        for (AiProvider provider : aiProviders) {
            providerMap.put(provider.getProviderType(), provider);
            log.info("注册AI服务提供商: {}", provider.getProviderType().getName());
        }
    }
    
    /**
     * 根据提供商类型获取AI服务实现
     * 
     * @param providerType 提供商类型
     * @return AI服务实现
     */
    public AiProvider getProvider(AiProviderEnum providerType) {
        AiProvider provider = providerMap.get(providerType);
        if (provider == null) {
            throw new IllegalArgumentException("不支持的AI提供商: " + providerType.getName());
        }
        return provider;
    }
    
    /**
     * 根据代码获取AI服务实现
     * 
     * @param code 提供商代码
     * @return AI服务实现
     */
    public AiProvider getProvider(String code) {
        AiProviderEnum providerType = AiProviderEnum.fromCode(code);
        return getProvider(providerType);
    }
    
    /**
     * 获取所有可用的提供商
     * 
     * @return 提供商映射
     */
    public Map<AiProviderEnum, AiProvider> getAllProviders() {
        return new HashMap<>(providerMap);
    }
}
